import jax
import jax.numpy as jnp
from cec2022 import CEC2022TestSuit

# 创建特定测试函数(1-12)
problem = CEC2022TestSuit.create(1)  # 创建 F1_CEC2022 实例

# 初始化随机密钥
key = jax.random.PRNGKey(0)

# 生成随机样本点
X = jax.random.uniform(key, shape=(1, 10)) * 200 - 100  # 在10D空间中生成100个点

# 正确初始化状态
state = problem.init(key)  # 使用init方法初始化状态

# 评估函数
F, state = problem.evaluate(state, X)

print(F)